# docker build -f spleeter-2.2.2-gpu.Dockerfile -t m986883511/spleeter:2.2.2-gpu .lsblk

FROM tensorflow/tensorflow:2.3.0-gpu
MAINTAINER author "https://github.com/m986883511"
WORKDIR /m986883511
EXPOSE 8000
ENV BIND_IP 0.0.0.0
ENV BIND_PORT 8000
ENV INPUT_DIR /input
ENV OUTPUT_DIR /output

ADD docker/ffmpeg.tar /usr/bin
ADD docker/pretrained_models.tar /pretrained_models

COPY docker/requirements.txt .
RUN pip3 install -r requirements.txt --no-cache-dir -i https://pypi.tuna.tsinghua.edu.cn/simple/
RUN pip3 install click==7.1.1 --no-cache-dir -i https://pypi.tuna.tsinghua.edu.cn/simple/
COPY . .
RUN rm -rf docker
CMD ["python3", "main.py"]